rflow (version 0.0.0.9000)

TensorFlow DNN Classifier: TensorFlow DNN Classifier

Description

Deep Neural Network classifier

Usage

TensorFlowDNNClassifier(hidden_units, n_classes, ...)

Arguments

hidden_units

A vector of the number of hidden units in each layer

n_classes

The number of classes in the target

...

Additional parameters such as keep_prob, a list can be found here: https://github.com/google/skflow/blob/master/skflow/ops/dnn_ops.py